-
Notifications
You must be signed in to change notification settings - Fork 5.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[CINN+PIR]Support DoGroupSchedule for PIRCompiler #58399
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
fix complation problem
return op_pattern_dict[cinn_op]; | ||
} | ||
|
||
std::vector<int> CompatibleInfo::ValueShape(const ::pir::Value& value) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
std::vector<int> CompatibleInfo::ValueShape(const ::pir::Value& value) { | |
std::vector<int> CompatibleInfo::ValueShape(::pir::Value value) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, 我下个PR一起fix
@@ -59,6 +61,12 @@ struct CompatibleInfo { | |||
static utils::AttributeMap ConvertAttributes(const ::pir::Operation& op); | |||
|
|||
static common::Type ConvertIRType(::pir::Type type); | |||
|
|||
static std::vector<int> ValueShape(const ::pir::Value& value); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
static std::vector<int> ValueShape(const ::pir::Value& value); | |
static std::vector<int> ValueShape(::pir::Value value); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, 我下个PR一起fix
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* [CINN+PIR]Support DoGroupSchedule for PIRComppiler fix complation problem * fix conflict * using output_ops to parse function arguments * fix unittest * remove VLOG(1) * ignore some UT and add FIXME
* [CINN+PIR]Support DoGroupSchedule for PIRComppiler fix complation problem * fix conflict * using output_ops to parse function arguments * fix unittest * remove VLOG(1) * ignore some UT and add FIXME
PR types
New features
PR changes
Others
Description
Pcard-67164
What's New?
DoGroupSchedule
优化策略功能op_lowering_utils.h/.cc
,解耦CINN现有的utils.cc,互不干扰Group
的数据结构,丰富了数据成员,方便与OpFusion/FusionMerge
打通softmax 子图生成代码:
遗留TODO项
CanInline
策略在一个单测上存在问题,并临时注释了2个Fusion相关的单测,待单独PR修复打开;OpFusion/FusionMerge
待切换为默认开启模式,并完成后端串通调试;